from tensorpack.dataflow import *

class MyCustomData(DataFlow):
    def __init__(self, ...): # 初始化函数可以根据需要修改参数列表
        pass
    
    def size(self):
        return 1000   # 返回 dataset 总数
        
    def get_data(self):
        for i in range(1000):
            yield [data_point]  # 替换为实际返回值
            
# 使用 PrefetchData 提升性能        
ds = PrefetchData(MyCustomData(), nr_prefetch=50)